package com.springcloud.servicezuul.config;

import com.netflix.zuul.ZuulFilter;
import com.netflix.zuul.context.RequestContext;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Component;

import javax.servlet.http.HttpServletRequest;

/**
 * @author chenkuan
 * @version v1.0
 * @desc 实现ZuulFilter过滤器
 * @date 2019/4/25 025 15:35
 */
@Slf4j
@Component
public class MyFilter extends ZuulFilter {

	/**
	 * filterType：返回一个字符串代表过滤器的类型，在zuul中定义了四种不同生命周期的过滤器类型，具体如下
	 * pre：路由之前
	 * routing：路由之时
	 * post： 路由之后
	 * error：发送错误调用
	 */
	@Override
	public String filterType() {
		return "pre";
	}

	/**
	 * filterOrder：过滤的顺序
	 */
	@Override
	public int filterOrder() {
		return 0;
	}

	/**
	 * shouldFilter：这里可以写逻辑判断，是否要过滤，本文true,永远过滤
	 */
	@Override
	public boolean shouldFilter() {
		return true;
	}

	/**
	 * run：过滤器的具体逻辑。可用很复杂，包括查sql，nosql去判断该请求到底有没有权限访问
	 */
	@Override
	public Object run() {
		RequestContext requestContext = RequestContext.getCurrentContext();
		HttpServletRequest request = requestContext.getRequest();
		log.info(String.format("%s >>> %s", request.getMethod(), request.getRequestURL().toString()));
		Object accessToken = request.getParameter("token");
		if (accessToken == null) {
			log.warn("token is empty");
			requestContext.setSendZuulResponse(false); // 过滤该请求，不对其进行路由
			requestContext.setResponseStatusCode(401); // 设置响应状态码
			try {
				requestContext.getResponse().getWriter().write("token is empty");
			} catch (Exception e) {
				log.error("zuul run Exception", e);
			}
			return null;
		}
		log.info("ok");
		return null;
	}

}
